How to tune neural network in python